import sqlite3

from sqlite3 import Error


# 数据库通用工具类
# 获取连接
def get_db_conn(db_file):
    conn = None
    # 下面的基本逻辑是：判断是否存在被连接的数据库，如果有就进行连接，如果没有则返回错误
    try:
        conn = sqlite3.connect(db_file)
    except Error as e:
        print(e)
    if conn is not None:
        return conn

# 关闭资源
def close_db_conn(cur, conn):
    if cur is not None:
        cur.close()
    if conn is not None:
        conn.close()
